package com.atguigu.gmall.common.config;

import com.alibaba.fastjson.JSON;
import com.atguigu.gmall.common.model.GmallCorrelationData;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.concurrent.TimeUnit;

/**
 * @Author atguigu-mqx
 * @Description 让消息服务器显示告诉我们消息是否的到达，交换机与对象
 * @Date 2023/5/26 11:40
 * @Version 1.0
 */
@Component
@Slf4j
public class MQProducerAckConfig implements RabbitTemplate.ReturnCallback,RabbitTemplate.ConfirmCallback {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    //  类似于初始化方法
    @PostConstruct
    public void init(){
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }
    /**
     * 判断消息是否正确到达交换机
     * @param correlationData
     * @param flag
     * @param replayText
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean flag, String replayText) {
        if (flag){
            System.out.println("消息正确到交换机");
        }else {
            System.out.println("消息没有到交换机"+replayText);
            //  调用重发方法.
            this.retrySendMsg(correlationData);
        }
    }



    /**
     * 判断消息是否到达队列：没有到达队列，这个方法才会被执行！
     * @param message
     * @param replyCode
     * @param replyText
     * @param exchange
     * @param routingKey
     */
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        log.error("消息路由queue失败，应答码={}，原因={}，交换机={}，路由键={}，消息={}",
                replyCode, replyText, exchange, routingKey, message.toString());
        //  message.getBody(); 消息主体; exchange routingKey ,没有记录当前是重发了几次; count

        //  重发 uuid: 发送消息的时候，将uuid 赋值了 spring_returned_message_correlation 这个key
        String uuid = (String) message.getMessageProperties().getHeaders().get("spring_returned_message_correlation");
        //  去缓存中获取数据
        String gmallJson = (String) this.redisTemplate.opsForValue().get(uuid);
        //  转换对象
        GmallCorrelationData gmallCorrelationData = JSON.parseObject(gmallJson, GmallCorrelationData.class);
        //  只要是延迟消息 就不执行下面代码.
        //        if ("exchange.delay".equals(exchange) && "routing.delay".equals(routingKey)){
        //            return;
        //        }
        //  如果是延迟消息，不执行重试方法。
        if (gmallCorrelationData.isDelay()){
            return;
        }
        //  调用重发方法.
        retrySendMsg(gmallCorrelationData);
    }

    /**
     * 重发方法.
     * @param correlationData
     */
    private void retrySendMsg(CorrelationData correlationData) {
        //  创建 GmallCorrelationData 对象
        GmallCorrelationData gmallCorrelationData = (GmallCorrelationData) correlationData;
        //  获取到重试次数
        int retryCount = gmallCorrelationData.getRetryCount();
        //  判断
        if (retryCount>3){
            System.out.println("重试次数已到："+retryCount);
            //  消息记录表；将未发送成功的信息，记录起来，后续通过人工排查，解决！ insert into msg value(ex,routingkey,msg);
            log.error("发生失败{}",gmallCorrelationData.getRoutingKey());
        } else {
            //  变量迭代
            retryCount++;
            //  将迭代的内容写回缓存.
            gmallCorrelationData.setRetryCount(retryCount);
            this.redisTemplate.opsForValue().set(gmallCorrelationData.getId(),JSON.toJSONString(gmallCorrelationData),10, TimeUnit.MINUTES);
            //  判断是否是延迟消息
            if (gmallCorrelationData.isDelay()){
                this.rabbitTemplate.convertAndSend(gmallCorrelationData.getExchange(),gmallCorrelationData.getRoutingKey(),gmallCorrelationData.getMessage(),(message)-> {
                    //  设置延迟时间
                    message.getMessageProperties().setDelay(gmallCorrelationData.getDelayTime()*1000);
                    return message;
                },gmallCorrelationData);
            } else {
                //  发送消息.
                this.rabbitTemplate.convertAndSend(gmallCorrelationData.getExchange(),gmallCorrelationData.getRoutingKey(),gmallCorrelationData.getMessage(),gmallCorrelationData);

            }
          }
    }
}
